/* USER CODE BEGIN Header */
/**
 ******************************************************************************
 * @file    usart.c
 * @brief   This file provides code for the configuration
 *          of the USART instances.
 ******************************************************************************
 * @attention
 *
 * Copyright (c) 2024 STMicroelectronics.
 * All rights reserved.
 *
 * This software is licensed under terms that can be found in the LICENSE file
 * in the root directory of this software component.
 * If no LICENSE file comes with this software, it is provided AS-IS.
 *
 ******************************************************************************
 */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "usart.h"

/* USER CODE BEGIN 0 */
#include "string.h"
#include "pc.h"
#include "rs485.h"

#define U1_RXBUFFSIZE 128
#define U3_RXBUFFSIZE 128

uint8_t U1_RxBuf[U1_RXBUFFSIZE];
uint8_t U3_RxBuf[U3_RXBUFFSIZE];
uint8_t tmpChr1;
uint8_t tmpChr3;
UartRxCtrl U1_RxCtrl = {
    .RxBuff = U1_RxBuf,
    .BuffSize = U1_RXBUFFSIZE,
    .RxCount = 0,
    .RxBegin = 0,
    .RxOK = 0};
UartRxCtrl U3_RxCtrl = {
    .RxBuff = U3_RxBuf,
    .BuffSize = U3_RXBUFFSIZE,
    .RxCount = 0,
    .RxBegin = 0,
    .RxOK = 0};
/* USER CODE END 0 */

UART_HandleTypeDef huart1;
UART_HandleTypeDef huart3;
DMA_HandleTypeDef hdma_usart1_tx;

/* USART1 init function */

void MX_USART1_UART_Init(void)
{

  /* USER CODE BEGIN USART1_Init 0 */

  /* USER CODE END USART1_Init 0 */

  /* USER CODE BEGIN USART1_Init 1 */

  /* USER CODE END USART1_Init 1 */
  huart1.Instance = USART1;
  huart1.Init.BaudRate = 115200;
  huart1.Init.WordLength = UART_WORDLENGTH_8B;
  huart1.Init.StopBits = UART_STOPBITS_1;
  huart1.Init.Parity = UART_PARITY_NONE;
  huart1.Init.Mode = UART_MODE_TX_RX;
  huart1.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart1.Init.OverSampling = UART_OVERSAMPLING_16;
  if (HAL_UART_Init(&huart1) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART1_Init 2 */

  // __HAL_UART_ENABLE_IT(&huart1, UART_IT_RXNE);
  
  /* USER CODE END USART1_Init 2 */

}
/* USART3 init function */

void MX_USART3_UART_Init(void)
{

  /* USER CODE BEGIN USART3_Init 0 */

  /* USER CODE END USART3_Init 0 */

  /* USER CODE BEGIN USART3_Init 1 */

  /* USER CODE END USART3_Init 1 */
  huart3.Instance = USART3;
  huart3.Init.BaudRate = 9600;
  huart3.Init.WordLength = UART_WORDLENGTH_8B;
  huart3.Init.StopBits = UART_STOPBITS_1;
  huart3.Init.Parity = UART_PARITY_NONE;
  huart3.Init.Mode = UART_MODE_TX_RX;
  huart3.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart3.Init.OverSampling = UART_OVERSAMPLING_16;
  if (HAL_UART_Init(&huart3) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART3_Init 2 */
  // __HAL_UART_ENABLE_IT(&huart3, UART_IT_RXNE);
  
  /* USER CODE END USART3_Init 2 */

}

void HAL_UART_MspInit(UART_HandleTypeDef* uartHandle)
{

  GPIO_InitTypeDef GPIO_InitStruct = {0};
  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspInit 0 */

  /* USER CODE END USART1_MspInit 0 */
    /* USART1 clock enable */
    __HAL_RCC_USART1_CLK_ENABLE();

    __HAL_RCC_GPIOA_CLK_ENABLE();
    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10     ------> USART1_RX
    */
    GPIO_InitStruct.Pin = GPIO_PIN_9;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_HIGH;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_INPUT;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    /* USART1 DMA Init */
    /* USART1_TX Init */
    hdma_usart1_tx.Instance = DMA1_Channel4;
    hdma_usart1_tx.Init.Direction = DMA_MEMORY_TO_PERIPH;
    hdma_usart1_tx.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_usart1_tx.Init.MemInc = DMA_MINC_ENABLE;
    hdma_usart1_tx.Init.PeriphDataAlignment = DMA_PDATAALIGN_BYTE;
    hdma_usart1_tx.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    hdma_usart1_tx.Init.Mode = DMA_NORMAL;
    hdma_usart1_tx.Init.Priority = DMA_PRIORITY_LOW;
    if (HAL_DMA_Init(&hdma_usart1_tx) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(uartHandle,hdmatx,hdma_usart1_tx);

    /* USART1 interrupt Init */
    HAL_NVIC_SetPriority(USART1_IRQn, 0, 0);
    HAL_NVIC_EnableIRQ(USART1_IRQn);
  /* USER CODE BEGIN USART1_MspInit 1 */

  /* USER CODE END USART1_MspInit 1 */
  }
  else if(uartHandle->Instance==USART3)
  {
  /* USER CODE BEGIN USART3_MspInit 0 */

  /* USER CODE END USART3_MspInit 0 */
    /* USART3 clock enable */
    __HAL_RCC_USART3_CLK_ENABLE();

    __HAL_RCC_GPIOB_CLK_ENABLE();
    /**USART3 GPIO Configuration
    PB10     ------> USART3_TX
    PB11     ------> USART3_RX
    */
    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_HIGH;
    HAL_GPIO_Init(GPIOB, &GPIO_InitStruct);

    GPIO_InitStruct.Pin = GPIO_PIN_11;
    GPIO_InitStruct.Mode = GPIO_MODE_INPUT;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    HAL_GPIO_Init(GPIOB, &GPIO_InitStruct);

    /* USART3 interrupt Init */
    HAL_NVIC_SetPriority(USART3_IRQn, 0, 0);
    HAL_NVIC_EnableIRQ(USART3_IRQn);
  /* USER CODE BEGIN USART3_MspInit 1 */

  /* USER CODE END USART3_MspInit 1 */
  }
}

void HAL_UART_MspDeInit(UART_HandleTypeDef* uartHandle)
{

  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspDeInit 0 */

  /* USER CODE END USART1_MspDeInit 0 */
    /* Peripheral clock disable */
    __HAL_RCC_USART1_CLK_DISABLE();

    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10     ------> USART1_RX
    */
    HAL_GPIO_DeInit(GPIOA, GPIO_PIN_9|GPIO_PIN_10);

    /* USART1 DMA DeInit */
    HAL_DMA_DeInit(uartHandle->hdmatx);

    /* USART1 interrupt Deinit */
    HAL_NVIC_DisableIRQ(USART1_IRQn);
  /* USER CODE BEGIN USART1_MspDeInit 1 */

  /* USER CODE END USART1_MspDeInit 1 */
  }
  else if(uartHandle->Instance==USART3)
  {
  /* USER CODE BEGIN USART3_MspDeInit 0 */

  /* USER CODE END USART3_MspDeInit 0 */
    /* Peripheral clock disable */
    __HAL_RCC_USART3_CLK_DISABLE();

    /**USART3 GPIO Configuration
    PB10     ------> USART3_TX
    PB11     ------> USART3_RX
    */
    HAL_GPIO_DeInit(GPIOB, GPIO_PIN_10|GPIO_PIN_11);

    /* USART3 interrupt Deinit */
    HAL_NVIC_DisableIRQ(USART3_IRQn);
  /* USER CODE BEGIN USART3_MspDeInit 1 */

  /* USER CODE END USART3_MspDeInit 1 */
  }
}

/* USER CODE BEGIN 1 */
void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
{
  static uint8_t frameLen = 255;
  if (huart->Instance == USART1)
  {
    if (U1_RxCtrl.RxCount == 0 && tmpChr1 == 0xE0)
    {
      U1_RxCtrl.RxBegin = 1;
      U1_RxCtrl.RxBuff[U1_RxCtrl.RxCount++] = tmpChr1;
    }
    else if (U1_RxCtrl.RxBegin == 1)
    {

      if (U1_RxCtrl.RxCount < frameLen)
      {
        U1_RxCtrl.RxBuff[U1_RxCtrl.RxCount++] = tmpChr1;
        if (U1_RxCtrl.RxCount == 3)
        {
          frameLen = tmpChr1 + 1;
        }
        if (U1_RxCtrl.RxCount == frameLen)
        {
          memcpy(CMD_FromPC, U1_RxCtrl.RxBuff, U1_RxCtrl.RxCount);
          // HAL_UART_Transmit(huart, U1_RxCtrl.RxBuff, U1_RxCtrl.RxCount, 100);
          U1_RxCtrl.RxBegin = 0;
          U1_RxCtrl.RxOK = 1;
          frameLen = 255;
          // Uart_ResetRx(U1_RxCtrl);
        }
      }
      //===================================
      // if (U1_RxCtrl.RxCount == 1)
      // {
      //   // frameLen = tmpChr1 + 1;
      //   U1_RxCtrl.RxBuff[U1_RxCtrl.RxCount++] = tmpChr1;
      // }
      // else if (U1_RxCtrl.RxCount == 2)
      // {
      //   // frameLen = tmpChr1 + 1;
      //   U1_RxCtrl.RxBuff[U1_RxCtrl.RxCount++] = tmpChr1;
      // }
      // else if (U1_RxCtrl.RxCount < U1_RxCtrl.RxBuff[2] + 1)
      // {
      //   U1_RxCtrl.RxBuff[U1_RxCtrl.RxCount++] = tmpChr1;
      // }
      // if (U1_RxCtrl.RxCount >= U1_RxCtrl.RxBuff[2] + 1)
      // {
      //   memcpy(CMD_FromPC, U1_RxCtrl.RxBuff, U1_RxCtrl.RxCount);
      //   HAL_UART_Transmit(huart, U1_RxCtrl.RxBuff, U1_RxCtrl.RxCount, 100);
      //   U1_RxCtrl.RxBegin = 0;
      //   U1_RxCtrl.RxOK = 1;
      //   // Uart_ResetRx(U1_RxCtrl);
      // }
    }
    else
    {
      U1_RxCtrl.RxCount = 0;
      U1_RxCtrl.RxBegin = 0;
      U1_RxCtrl.RxOK = 0;
    }
    HAL_UART_Receive_IT(huart, &tmpChr1, 1);
  }
  else if (huart->Instance == USART3)
  {
    if (U3_RxCtrl.RxCount == 0 && tmpChr3 == 0xE0)
    {
      // 如果U3数据已经处理完成
      if (U3_RxCtrl.RxOK == 0)
      {
        U3_RxCtrl.RxBegin = 1;
        U3_RxCtrl.RxBuff[U3_RxCtrl.RxCount++] = tmpChr3;
      }
    }
    else if (U3_RxCtrl.RxBegin == 1)
    {
      if (U3_RxCtrl.RxCount == 1)
      {
        // frameLen = tmpChr1 + 1;
        U3_RxCtrl.RxBuff[U3_RxCtrl.RxCount++] = tmpChr3;
      }
      else if (U3_RxCtrl.RxCount < U3_RxCtrl.RxBuff[1] + 1)
      {
        U3_RxCtrl.RxBuff[U3_RxCtrl.RxCount++] = tmpChr3;
      }
      if (U3_RxCtrl.RxCount >= U3_RxCtrl.RxBuff[1] + 1)
      {
        memcpy(_485_CMDRecv, U3_RxCtrl.RxBuff, U3_RxCtrl.RxCount);
        // HAL_UART_Transmit(huart, U3_RxCtrl.RxBuff, U3_RxCtrl.RxCount, 100);
        U3_RxCtrl.RxBegin = 0;
        U3_RxCtrl.RxOK = 1;
        // Uart_ResetRx(U3_RxCtrl);
      }
    }
    else
    {
      U3_RxCtrl.RxCount = 0;
      U3_RxCtrl.RxBegin = 0;
      U3_RxCtrl.RxOK = 0;
    }
    HAL_UART_Receive_IT(huart, &tmpChr3, 1);
  }
}
/* USER CODE END 1 */
